#include "mystdio.h"

#define FILE_MODE 0666

_FILE* _fopen(char* filename, const char* flag)
{
    assert(filename && flag);
    int fd = -1;
    if(strcmp(flag, "w") == 0)
    {
        fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, FILE_MODE);

    }
    else if(strcmp(flag, "a") == 0)
    {
        fd = open(filename, O_WRONLY|O_CREAT|O_APPEND, FILE_MODE);
    }
    else if(strcmp(flag, "r") == 0)
    {
        fd = open(filename, O_RDONLY);
    }
    else 
    {
        return NULL;
    }
    
    if(fd == -1) return NULL;

    _FILE* fp = (_FILE*)malloc(sizeof(_FILE));
    if(fp == NULL) return NULL;

    fp->fileno = fd;
    fp->flag = FLUSH_ALL;
    fp->out_pos = 0;

    return fp;
}

ssize_t _fwrite(_FILE *fp, const void *s, size_t len)
{
    memcpy(&fp->outbuffer[fp->out_pos], s, len);//未处理异常情况，如空间不足因素（动态内存管理可解决）
    fp->out_pos += len;

    ssize_t size = 0;
    if(fp->flag&FLUSH_NOW)
    {
        size = write(fp->fileno, fp->outbuffer, fp->out_pos);
        fp->out_pos = 0;
    }
    else if(fp->flag&FLUSH_LINE)
    {
        if(fp->outbuffer[fp->out_pos-1] == '\n')//未考虑其他刷新情况
        {
            size = write(fp->fileno, fp->outbuffer, fp->out_pos);
            fp->out_pos = 0;
        }
    }
    else if(fp->flag&FLUSH_ALL)
    {
        if(fp->out_pos == SIZE)
        {
            size = write(fp->fileno, fp->outbuffer, fp->out_pos);
            fp->out_pos = 0;
        }
    }

    return size;
}

void _fflush(_FILE* _fp)
{
    if(_fp->out_pos > 0)
    {
        write(_fp->fileno, _fp->outbuffer, _fp->out_pos);
        _fp->out_pos = 0;
    }
}

void _fclose(_FILE* fp)
{
    if(fp == NULL) return;
    //关闭文件前，刷新一下缓冲区
    _fflush(fp);
    close(fp->fileno);
    free(fp);
}
